package com.heima.wemedia.service.impl;


import com.alibaba.fastjson.JSON;
import com.heima.aliyun.scan.GreenScan;
import com.heima.aliyun.scan.ScanResult;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.constants.message.PublishArticleConstants;
import com.heima.model.common.constants.wemedia.WemediaConstants;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.*;
import java.util.stream.Collectors;

@Slf4j
@Service
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {
    @Value("${file.oss.web-site}")
    private String webSite;
    @Autowired
    private WmNewsMapper wmNewsMapper;

    @Autowired
    private GreenScan greenScan;
    @Autowired
    private AdminFeign adminFeign;

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Override
    public void autoScanWmNews(Integer id) {
        log.info("文章自动审核方法触发 待审核的文章id: {}", id);
        //1.判断文章id是否为空
        if (id == null) {
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID, "待审核文章的id为空");

        }

        //2.根据id查询自媒体文章
        WmNews wmNews = wmNewsMapper.selectById(id);
        if (wmNews == null) {
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST, "对应的文章不存在");

        }

        //判断文章的 状态 必须是1 (待审核)

        Short status = wmNews.getStatus();
        if (!WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(status)) {
            CustException.cust(AppHttpCodeEnum.NO_OPERATOR_AUTH, "当前文章无需审核");
        }

        //4.抽取文章中所有的文本内容 和 所有的图片内容
        Map<String, Object> contentAndImages = handleTextAndImages(wmNews);

        //5.DFA 进行滋味哦盖里敏感词审核  2 有敏感词不通过  通过继续下一步

        boolean scanSensitive = handleSensitive((String) contentAndImages.get("content"), wmNews);

        if (!scanSensitive) {
            CustException.cust(AppHttpCodeEnum.DATA_NOT_ALLOW, "文章审核未通过,原因:内容中包含敏感词");
        }
        // 6. 阿里云的文本审核  2 有违规词汇    3 不确定/aliyun未调用成功   通过继续下一步
        boolean scanText = handleTextScan((String) contentAndImages.get("content"), wmNews);

        if (!scanText) {
            //返回false 代表审核不成功  结束处理
            CustException.cust(AppHttpCodeEnum.DATA_NOT_ALLOW, "内容中包含违规词汇");
        }
        // 7. 阿里云的图片审核  2 有违规图片    3 不确定/aliyun未调用成功   通过继续下一步
        boolean scanImages = handleImageScan((List<String>) contentAndImages.get("images"), wmNews);
        if (!scanImages) {
            //返回false代表审核不成功  结束处理
            CustException.cust(AppHttpCodeEnum.DATA_NOT_ALLOW, "图片中包含违规词汇");
        }
        //8.将文章状态改为 8 自动审核通过
        updateWmNews(WmNews.Status.SUCCESS.getCode(), "审核通过", wmNews);
        // 9. TODO 根据文章的发布时间，发送延迟消息 用于定时发布文章( 9 已发布)

        // 发布时间
        long publishTime = wmNews.getPublishTime().getTime();
        // 当前时间
        long nowTime = new Date().getTime();
        long remainTime = publishTime - nowTime;
        // 发布文章
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE
                , PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY
                , wmNews.getId()
                , (message) -> {                              // 延时消息 必设置
                    message.getMessageProperties().setHeader("x-delay", remainTime <= 0 ? 0 : remainTime);
                    return message;
                }
        );
        log.info("立即发布文章通知成功发送，文章id : {}", wmNews.getId());
    }

    /**
     * @param images 待审核的图片
     * @param wmNews 文章
     * @return
     */
    private boolean handleImageScan(List<String> images, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.imageUrlScan(images);
            switch (scanResult.getSuggestion()) {
                case "block":
                    // 失败
                    updateWmNews(WmNews.Status.FAIL.getCode(), "文章图片中违规信息: " + scanResult.getLabel(), wmNews);
                    break;
                case "review":
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "文章图片中有不确定因素，需要进一步人工审核", wmNews);
                    break;
                case "pass":
                    // 成功
                    flag = true;
                    break;
                default:
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用状态异常，需要进一步人工审核:", wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 人工审核
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用失败，需要进一步人工审核:", wmNews);
        }
        return flag;
    }

    /**
     * 待审核文本
     *
     * @param content
     * @param wmNews
     * @return
     */
    private boolean handleTextScan(String content, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.greenTextScan(content);
            //阿里云建一: block 违规  review 进一步人工审核  pass通过
            String suggestion = scanResult.getSuggestion();
            switch (suggestion) {
                case "block":
                    // 失败
                    updateWmNews(WmNews.Status.FAIL.getCode(), "文章内容中包含违规词汇: " + scanResult.getLabel(), wmNews);
                    break;
                case "review":
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "文章内容中有不确定因素，需要进一步人工审核:", wmNews);
                    break;
                case "pass":
                    // 成功
                    flag = true;
                    break;
                default:
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用状态异常，需要进一步人工审核:", wmNews);
                    break;
            }

        } catch (Exception e) {
            e.printStackTrace();
            // 人工
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用失败，需要进一步人工审核:", wmNews);
        }
        return flag;
    }


    /**
     * 待审核的文本内容
     *
     * @param content
     * @param wmNews
     * @return
     */
    private boolean handleSensitive(String content, WmNews wmNews) {
        boolean flag = true;
        //1.远程查询敏感词列表
        // TODO   先查询redis  看redis中是否缓存敏感词数据
        // TODO   如果redis没有  才使用远程查询admin服务
        ResponseResult<List<String>> sensitives = adminFeign.sensitives();
        if (!sensitives.checkCode()) {
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR);
        }
        List<String> sensitivesData = sensitives.getData();
        //2.将敏感词转为DFA 数据模型
        SensitiveWordUtil.initMap(sensitivesData);

        //3.基于DFA扫描 内容中是否包含敏感词
        Map<String, Integer> map = SensitiveWordUtil.matchWords(content);
        //4.如果有敏感词,将文章状态修改为 2  并告知原因
        if (!CollectionUtils.isEmpty(map)) {
            updateWmNews(WmNews.Status.FAIL.getCode(), "文章中包含敏感词 :" + map, wmNews);
            flag = false;
        }
        return flag;
    }

    private void updateWmNews(Short status, String reason, WmNews wmNews) {
        wmNews.setStatus(status);
        wmNews.setReason(reason);
        wmNewsMapper.updateById(wmNews);
    }

    /**
     * 抽取文章中所有的文本内容 和图片内容
     *
     * @param wmNews 文章
     * @return
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {
        HashMap<String, Object> result = new HashMap<>();
        //1.判断内容不能为空  并转为List<Map>
        String content = wmNews.getContent();
        if (StringUtils.isBlank(content)) {
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID, "文章不能为空");

        }
        List<Map> contentMapList = JSON.parseArray(content, Map.class);
        //2.抽取文章中所有的文本内容
        //2.1抽取content 中所有文本并凭借成一个字符串
        String collect = contentMapList.stream()
                .filter(map -> "text".equals(map.get("type")))
                .map(map -> map.get("value").toString())
                .collect(Collectors.joining("_hmtt_"));

        //2.2将文本内容和标题凭借成一个字符串
        collect = wmNews.getTitle() + "_hmtt_" + collect;
        //2.3将总的文本内容装入 map
        result.put("content", collect);
        //3. 抽取文庄中所有的图片列表
        //3.1 中抽取文章中所有的图片列表
        List<String> images = contentMapList.stream()
                .filter(map -> "image".equals(map.get("type")))
                .map(map -> map.get("value").toString())
                .collect(Collectors.toList());

        //3.2抽取封面中所有的图片的图片列表
        String coverStr = wmNews.getImages();
        if (StringUtils.isNotBlank(coverStr)) {
            List<String> coverImages = Arrays.stream(coverStr.split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());

            //3.3 合并 内容图片 和 封面图片
            images.addAll(coverImages);
        }
        //3.4 去重幅图片
        images = images.stream().distinct().collect(Collectors.toList());
        //3.5将所有图片 装入map
        result.put("images", images);
        return result;
    }
}
